'use strict'

var _interopRequireDefault = require('@babel/runtime/helpers/interopRequireDefault')

Object.defineProperty(exports, '__esModule', {
  value: true
})
exports.spyElementPrototype = spyElementPrototype
exports.spyElementPrototypes = spyElementPrototypes

var _defineProperty2 = _interopRequireDefault(require('@babel/runtime/helpers/defineProperty'))

var _extends2 = _interopRequireDefault(require('@babel/runtime/helpers/extends'))

var __NULL__ = {
  notExist: true
}

function spyElementPrototypes(Element, properties) {
  var propNames = Object.keys(properties)
  var originDescriptors = {}
  propNames.forEach(function(propName) {
    var originDescriptor = Object.getOwnPropertyDescriptor(Element.prototype, propName)
    originDescriptors[propName] = originDescriptor || __NULL__
    var spyProp = properties[propName]

    if (typeof spyProp === 'function') {
      // If is a function
      Element.prototype[propName] = function spyFunc() {
        for (var _len = arguments.length, args = new Array(_len), _key = 0; _key < _len; _key++) {
          args[_key] = arguments[_key]
        }

        return spyProp.call.apply(spyProp, [this, originDescriptor].concat(args))
      }
    } else {
      // Otherwise tread as a property
      Object.defineProperty(Element.prototype, propName, (0, _extends2.default)((0, _extends2.default)({}, spyProp), {
        set: function set(value) {
          if (spyProp.set) {
            return spyProp.set.call(this, originDescriptor, value)
          }

          return originDescriptor.set(value)
        },
        get: function get() {
          if (spyProp.get) {
            return spyProp.get.call(this, originDescriptor)
          }

          return originDescriptor.get()
        }
      }))
    }
  })
  return {
    mockRestore: function mockRestore() {
      propNames.forEach(function(propName) {
        var originDescriptor = originDescriptors[propName]

        if (originDescriptor === __NULL__) {
          delete Element.prototype[propName]
        } else if (typeof originDescriptor === 'function') {
          Element.prototype[propName] = originDescriptor
        } else {
          Object.defineProperty(Element.prototype, propName, originDescriptor)
        }
      })
    }
  }
}

function spyElementPrototype(Element, propName, property) {
  return spyElementPrototypes(Element, (0, _defineProperty2.default)({}, propName, property))
}
